# 修复上传目录结构的脚本
import os
import shutil

# 定义路径
upload_dir = 'c:\\Users\\25188\\Downloads\\git\\backend\\uploads'
txt_file = os.path.join(upload_dir, 'txt')
new_txt_dir = os.path.join(upload_dir, 'txt')
backup_file = os.path.join(upload_dir, 'txt_backup.txt')

# 检查是否存在名为'txt'的文件
if os.path.exists(txt_file) and os.path.isfile(txt_file):
    print(f'发现名为txt的文件: {txt_file}')
    
    # 备份该文件
    print(f'备份文件到: {backup_file}')
    shutil.move(txt_file, backup_file)
    print('文件备份完成')
    
    # 创建txt目录
    if not os.path.exists(new_txt_dir):
        print(f'创建txt目录: {new_txt_dir}')
        os.makedirs(new_txt_dir)
        print('txt目录创建完成')
    else:
        print(f'txt目录已存在: {new_txt_dir}')
else:
    print(f'未找到名为txt的文件或它不是一个文件: {txt_file}')
    
    # 检查是否存在txt目录
    if os.path.exists(new_txt_dir) and os.path.isdir(new_txt_dir):
        print(f'txt目录已存在且是一个有效的目录: {new_txt_dir}')
    else:
        print(f'创建txt目录: {new_txt_dir}')
        os.makedirs(new_txt_dir)
        print('txt目录创建完成')

print('上传目录结构修复完成！')
print('现在您可以重新上传文件并测试RAG功能。')